# 导入蓝图对象
from flask import session, render_template, current_app, jsonify, request
# 导入模型类
from info.models import User, News, Category
# 导入自定义的状态码
from info.utils.response_code import RET
from . import news_blue


# 项目首页
@news_blue.route('/')
def index():
    """
    用户信息的展示
    1、尝试从redis中获取用户id
    2、如果user_id存在，查询mysql，把用户信息传给模板
    :return:
    """
    user_id = session.get('user_id')
    data = None
    user = None
    if user_id:
        try:
            user = User.query.get(user_id)
        except Exception as e:
            current_app.logger.error(e)
            return jsonify(errno=RET.DBERR, errmsg='查询数据失败')
            # if user:
            #     user.to_dict()
            # else:
            #     user = None
    # 新闻点击排行
    try:
        # 根据新闻点击量进行排序，并取出前六条数据，出入news_list
        # News.query.filter().order_by(News.clicks.desc()).limit(6)
        news_list = News.query.order_by(News.clicks.desc()).limit(6)
    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno=RET.DBERR, errmsg='查询新闻数据异常')
    # 判断查询结果是否存在
    if not news_list:
        return jsonify(errno=RET.NODATA, errmsg='新闻数据不存在')
    # 遍历查询结果，将查询结果作为字典，保存在列表中
    news_dict_list = []
    for news in news_list:
        news_dict_list.append(news.to_dict())

    data = {
        'user_info': user.to_dict() if user else None,
        'news_dict_list': news_dict_list
    }

    # 首页新闻分类数据展示
    try:
        categories = Category.query.all()
    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno=RET.DBERR, errmsg='查询新闻分类数据异常')
    # 判断查询结果
    if not categories:
        return jsonify(errno=RET.NODATA, errmsg='无新闻分类数据')
    # 定义容器，遍历查询结果
    category_list = []
    for category in categories:
        category_list.append(category.to_dict())

    data = {
        'user_info': user.to_dict() if user else None,
        'news_dict_list': news_dict_list,
        'category_list': category_list
    }
    return render_template('news/index.html', data=data)


@news_blue.route('/news_list')
def get_news_list():
    """
    首页新闻列表数据
    1、获取参数，cid/page/per_page
    2、检查参数：把参数转成int类型
    3、判断分类id是大于1
    4、根据新闻分类ID查询数据库，按照新闻发布时间进行排序，对排序结果进行分页
    5、获取分页后的新闻数据：总页数、当前页数、数据
    6、返回结果

    :return:
    """
    # 获取参数(如果不存在，则默认值是1)
    cid = request.args.get("cid", '1')
    page = request.args.get('page', '1')
    per_page = request.args.get("per_page", '10')
    # 检查参数，需要把参数转成int
    try:
        cid, page, per_page = int(cid), int(page), int(per_page)
    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno=RET.PARAMERR, errmsg='参数的格式错误')
    # 定义过滤条件
    filters = []
    # 判断分类id是否是最新
    if cid > 1:
        filters.append(News.category_id == cid)
    # 根据分类id查询数据库
    try:
        # 对过滤查询执行排序，对排序结果进行分页
        paginate = News.query.filter(*filters).order_by(News.create_time.desc()).paginate(page, per_page, False)
    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno=RET.DBERR, errmsg='查询新闻列表数据异常')
    # 获取分页后的新闻数据,总页数，当前页数
    news_list = paginate.items
    total_page = paginate.pages
    current_page = paginate.page
    # 定义列表
    news_dict_list = []
    # 遍历分页后的新闻数据
    for news in news_list:
        news_dict_list.append(news.to_dict())
    # 定义data字典
    data = {
        'news_dict_list': news_dict_list,
        'total_page': total_page,
        'current_page': current_page
    }
    # 返回结果，并传参。
    return jsonify(errno=RET.OK, errmsg='OK', data=data)


# 项目logo图标展示
@news_blue.route('/favicon.ico')
def favicon():
    # 使用current_app调用发送静态文件的方法，把项目logo文件发给浏览器
    return current_app.send_static_file('news/favicon.ico')
